#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
@Time    : 2025/4/18 15:10
@Author  : HZP
@File    : 2.Model批处理.py
"""
from dotenv import load_dotenv
from langchain_core.messages import SystemMessage
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

# 加载配置文件
load_dotenv()
# 初始化模型
llm = ChatOpenAI(
    temperature=0.6,
    model="qwq"
)
# 加入提示语模版
system = SystemMessage("请根据以下要求进行回答")
prompt = ChatPromptTemplate.from_messages([
    ("system", "你是一个很厉害的程序员,请回答用户的问题."),
    ("user", "{answer}")
])
ai_message = llm.batch([
    prompt.invoke({"answer": "现在几点"}),
    prompt.invoke({"answer": "你所在地址"}),
])

for message in ai_message:
    print(message)
